
import os
from dotenv import load_dotenv

load_dotenv()

class Config:

    # 大模型配置
    LLM_URI = os.getenv("LLM_URI","")
    LLM_API_KEY = os.getenv("LLM_API_KEY", "")
    LLM_MODEL = os.getenv("LLM_MODEL")
    USE_THINKING = os.getenv("USE_THINKING", False)

    # 智能问数服务
    DATA_SERVER_URI = os.getenv("DATA_SERVER_URI","")

    # 向量模型
    EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL","")

    # 网络检索
    SERPER_API_KEY = os.getenv("SERPER_API_KEY", "")
    TAVILY_API_KEY = os.getenv("TAVILY_API_KEY", "")

    # 迭代次数
    MAX_ITERATIONS = os.getenv("MAX_ITERATIONS", 5)